#include "Headers/UI/roomdialog.h"
#include "ui_roomdialog.h"

#include "../../Headers/UI/roomeditdialog.h"

RoomDialog::RoomDialog(QWidget *parent) :
    QDialog(parent),
    ui(new Ui::RoomDialog)
{
    ui->setupUi(this);

    this->setWindowFlags(Qt::Dialog |
                         Qt::WindowMinimizeButtonHint |
                         Qt::WindowMaximizeButtonHint);

    QTextCodec::setCodecForCStrings(QTextCodec::codecForName("UTF-8"));

    Update();
}

RoomDialog::~RoomDialog()
{
    delete ui;
}

/**
  Закрыть окно
  */
void RoomDialog::on_closeButton_clicked()
{
    this->close();
}

/**
  Создать комнату
  */
void RoomDialog::on_createButton_clicked()
{
    RoomEditDialog createDialog;
    createDialog.exec();

    Update();
}

/**
  Изменить комнату
  */
void RoomDialog::on_editButton_clicked()
{
    if (checkSelected())
    {
        RoomEditDialog editDialog(MyUtils::Editing, getCurrentRoom());
        editDialog.setWindowTitle(trUtf8("Изменение номера"));
        editDialog.exec();

        Update();
    }
}

/**
  Удалить комнату
  */
void RoomDialog::on_deleteButton_clicked()
{
    if (checkSelected())
    {
        Room * room = getCurrentRoom();
        if (room != NULL)
        {
            QMessageBox message;
            message.setText(trUtf8("Вы действительно хотите удалить тип комнаты ") + room->getName() + trUtf8("?"));
            message.setStandardButtons(QMessageBox::Ok | QMessageBox::Cancel);
            message.setDefaultButton(QMessageBox::Ok);
            int result = message.exec();
            if (result == QMessageBox::Ok)
            {
                RoomDataAccessor dataRoom;
                dataRoom.deleteRoom(*room);
                Update();
            }
        }
        delete room;
    }
}

void RoomDialog::Update()
{
    RoomDataAccessor roomData;
    QSqlQueryModel * model = roomData.getRoomQueryModel();
    model->setHeaderData(0, Qt::Horizontal, trUtf8("id"));
    model->setHeaderData(1, Qt::Horizontal, trUtf8("Номер"));
    model->setHeaderData(2, Qt::Horizontal, trUtf8("Название"));
    model->setHeaderData(3, Qt::Horizontal, trUtf8("Этаж"));
    model->setHeaderData(4, Qt::Horizontal, trUtf8("Тип номера"));
    model->setHeaderData(5, Qt::Horizontal, trUtf8("Цена"));
    ui->tableView->setModel(model);
    ui->tableView->hideColumn(0);
    ui->tableView->horizontalHeader()->setResizeMode(QHeaderView::Stretch);
}

Room * RoomDialog::getCurrentRoom()
{
    int row = ui->tableView->currentIndex().row();
    int id = ui->tableView->model()->data(ui->tableView->model()->index(row, 0)).toInt();
    RoomDataAccessor dataRoom;
    return dataRoom.getRoom(id);
}

void RoomDialog::on_tableView_doubleClicked(const QModelIndex &index)
{
    on_editButton_clicked();
}

bool RoomDialog::checkSelected()
{
    if (ui->tableView->currentIndex().isValid())
    {
        return true;
    }
    else
    {
        QMessageBox msg;
        msg.setIcon(QMessageBox::Information);
        msg.setWindowTitle(trUtf8("Ошибка"));
        msg.setText(trUtf8("Пожалуйста выберите запись для того, чтобы изменить её"));
        msg.exec();
        return false;
    }
}

void RoomDialog::keyPressEvent(QKeyEvent *event)
{
    switch(event->key())
    {
    case Qt::Key_Delete:
        on_deleteButton_clicked();
        break;
    case Qt::Key_Escape:
        this->close();
        break;
    }
}
